graph representation using python